import pandas as pd
# from utils.databaseIo import DatabaseIo
# from globalConst import DataBaseOperateType, DataBaseQuery

def get_ml_100k():
    # 数据加载
    train_data = pd.read_csv('../DGL/ml/ml-1k.csv', sep=',')

    return train_data

# def get_fshl():
#     dbHandle = DatabaseIo()
#     if not dbHandle:
#         return None
#
#     # sql_user = """SELECT id,sex FROM account5000"""
#     sql_user = DataBaseQuery["dgl_user_info"]
#     # sql_course = "select id , system_course_id ,course_name from course_info"
#     # sql_course = """select id,classify_id from course5000"""
#     sql_course = DataBaseQuery["feature_classify"]
#     # sql_user = """select user_id from user_basic_info"""
#     # sql_dr = """select * from course_dr5000"""
#     sql_dr = DataBaseQuery["course_dr"]
#
#     result_dr = dbHandle.doSql(execType=DataBaseOperateType.SearchMany,
#                                sql=sql_dr)
#     result_course = dbHandle.doSql(execType=DataBaseOperateType.SearchMany,
#                                    sql=sql_course)
#     result_user = dbHandle.doSql(execType=DataBaseOperateType.SearchMany,
#                                  sql=sql_user)
#     print(result_dr)
#
#     # 数据加载
#     train_data = pd.DataFrame(list(result_dr))
#     train_data.columns = ['user_id', 'item_id', 'rating']
#     user_data = pd.DataFrame(list(result_user))
#     item_data = pd.DataFrame(list(result_course))
#
#     return train_data,user_data,item_data

# def get_bigraph():
#     dbHandle = DatabaseIo()
#     if not dbHandle:
#         return None
#
#     sql_user = DataBaseQuery["dgl_user_info"]
#
#     sql_course = DataBaseQuery["feature_classify"]
#
#     result_course = dbHandle.doSql(execType=DataBaseOperateType.SearchMany,
#                                    sql=sql_course)
#     result_user = dbHandle.doSql(execType=DataBaseOperateType.SearchMany,
#                                  sql=sql_user)
#     # 数据加载
#     train_data = pd.read_csv('../file_saved/toGcn.csv', header=None,
#                              names=['user_id', 'item_id', 'rating'])
#     print(train_data)
#     train_data.columns = ['user_id', 'item_id', 'rating']
#     user_data = pd.DataFrame(list(result_user))
#     item_data = pd.DataFrame(list(result_course))
#
#     return train_data, user_data, item_data



